Exercise 3.1 complete